package Draft;

import java.util.ArrayList;
import java.util.List;

public class pacific_atlantic_water_flow {
    int m , n ;
    int [][] height;
    int[] dx = { 1 , -1 , 0 , 0};
    int[] dy ={ 0 , 0 , -1 , 1};
    boolean[][] pac ,atl;
    public List<List<Integer>> pacificAtlantic(int[][] _heights) {
        height = _heights;
        m = height.length; n = height[0].length;
        pac = new boolean[m][n];
        atl = new boolean[m][n];

        // 太平洋
        for(int i = 0 ; i < m; i++)dfs(i , 0 , pac);
        for(int j = 0 ; j < n ; j++)dfs(0 , j , pac);

        // 大西洋
        for(int i = 0 ; i < m ; i++)dfs(i , n-1 , atl);
        for(int j = 0 ; j < n ; j++) dfs(m-1, j , atl);

        // 提取数据
        List<List<Integer>> ret = new ArrayList<>();
        for(int i = 0 ; i< m ;i++){
            for(int j = 0 ; j< n ; j++){
                if(pac[i][j] && atl[i][j]){
                    List<Integer> tmp = new ArrayList<>();
                    tmp.add(i); tmp.add(j);
                    ret.add(tmp);
                }
            }
        }
        return ret;

    }

    public void dfs(int i , int j , boolean[][] vis){
        vis[i][j]=true;
        for(int k = 0 ; k < 4 ; k++){
            int x = i +dx[k];
            int y = j + dy[k];
            if(x >= 0 && x < m && y >= 0 && y < n && !vis[x][y] && height[i][j] <= height[x][y]){
                dfs( x , y , vis);
            }
        }
    }
}
